/*****************************************************************************
* | File      	:   EPD_1IN54_A05.C
* | Author      :   Waveshare team
* | Function    :   1.54inch e-paper
* | Info        :
*----------------
* |	This version:   V3.0
* | Date        :   2019-06-12
* | Info        :
* -----------------------------------------------------------------------------
* V3.0(2019-06-12):
* 1.Change:
*    lut_full_update[] => EPD_1IN54_A05_lut_full_update[]
*    lut_partial_update[] => EPD_1IN54_A05_lut_partial_update[]
*    EPD_Reset() => EPD_1IN54_A05_Reset()
*    EPD_SendCommand() => EPD_1IN54_A05_SendCommand()
*    EPD_SendData() => EPD_1IN54_A05_SendData()
*    EPD_WaitUntilIdle() => EPD_1IN54_A05_ReadBusy()
*    EPD_SetLut() => EPD_1IN54_A05_SetLut()
*    EPD_SetWindow() => EPD_1IN54_A05_SetWindow()
*    EPD_SetCursor() => EPD_1IN54_A05_SetCursor()
*    EPD_TurnOnDisplay() => EPD_1IN54_A05_TurnOnDisplay()
*    EPD_Init() => EPD_1IN54_A05_Init()
*    EPD_Clear() => EPD_1IN54_A05_Clear()
*    EPD_Display() => EPD_1IN54_A05_Display()
*    EPD_Sleep() => EPD_1IN54_A05_Sleep()
* 2.remove commands define:
*   #define PANEL_SETTING                               0x00
*   #define POWER_SETTING                               0x01
*   #define POWER_OFF                                   0x02
*   #define POWER_OFF_SEQUENCE_SETTING                  0x03
*   #define POWER_ON                                    0x04
*   #define POWER_ON_MEASURE                            0x05
*   #define BOOSTER_SOFT_START                          0x06
*   #define DEEP_SLEEP                                  0x07
*   #define DATA_START_TRANSMISSION_1                   0x10
*   #define DATA_STOP                                   0x11
*   #define DISPLAY_REFRESH                             0x12
*   #define DATA_START_TRANSMISSION_2                   0x13
*   #define PLL_CONTROL                                 0x30
*   #define TEMPERATURE_SENSOR_COMMAND                  0x40
*   #define TEMPERATURE_SENSOR_CALIBRATION              0x41
*   #define TEMPERATURE_SENSOR_WRITE                    0x42
*   #define TEMPERATURE_SENSOR_READ                     0x43
*   #define VCOM_AND_DATA_INTERVAL_SETTING              0x50
*   #define LOW_POWER_DETECTION                         0x51
*   #define TCON_SETTING                                0x60
*   #define TCON_RESOLUTION                             0x61
*   #define SOURCE_AND_GATE_START_SETTING               0x62
*   #define GET_STATUS                                  0x71
*   #define AUTO_MEASURE_VCOM                           0x80
*   #define VCOM_VALUE                                  0x81
*   #define VCM_DC_SETTING_REGISTER                     0x82
*   #define PROGRAM_MODE                                0xA0
*   #define ACTIVE_PROGRAM                              0xA1
*   #define READ_OTP_DATA                               0xA2
* -----------------------------------------------------------------------------
* V2.0(2018-10-30):
* 1.Remove:ImageBuff[EPD_1IN54_A05_HEIGHT * EPD_1IN54_A05_WIDTH / 8]
* 2.Change:EPD_Display(UBYTE *Image)
*   Need to pass parameters: pointer to cached data
* 3.Change:
*   EPD_RST -> EPD_RST_PIN
*   EPD_DC -> EPD_DC_PIN
*   EPD_CS -> EPD_CS_PIN
*   EPD_BUSY -> EPD_BUSY_PIN
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documnetation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to  whom the Software is
# furished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS OR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
******************************************************************************/
#include "EPD_1IN54_A05.h"
#include "Debug.h"

static const unsigned char EPD_1IN54_A05_lut_full_update[] = {
    // 0x66, // machine LUT
    // 0x66, 0x44, 0x66, 0xAA, 0x11, 0x80, 0x08, 0x11, 0x18, 0x81,
    // 0x18, 0x11, 0x88, 0x11, 0x88, 0x11, 0x88, 0x00, 0x00, 0xFF,
    // 0xFF, 0xFF, 0xFF, 0x5F, 0xAF, 0xFF, 0xFF, 0x2F, 0x00
    0x02, 0x02, 0x01, 0x11, 0x12, 0x12, 0x22, 0x22,
    0x66, 0x69, 0x69, 0x59, 0x58, 0x99, 0x99, 0x88,
    0x00, 0x00, 0x00, 0x00, 0xF8, 0xB4, 0x13, 0x51,
    0x35, 0x51, 0x51, 0x19, 0x01, 0x00
    
    };



static const unsigned char EPD_1IN54_A05_lut_partial_update[] = {

    0x10, 0x18, 0x18, 0x28, 0x18, 0x18, 0x18, 0x18, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    0x00, 0x00, 0x13, 0x11, 0x22, 0x63, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00};

/******************************************************************************
function :	Software reset
parameter:
******************************************************************************/
static void EPD_1IN54_A05_Reset(void)
{
    DEV_Digital_Write(EPD_RST_PIN, 1);
    DEV_Delay_ms(200);
    DEV_Digital_Write(EPD_RST_PIN, 0);
    DEV_Delay_ms(10);
    DEV_Digital_Write(EPD_RST_PIN, 1);
    DEV_Delay_ms(200);
}

/******************************************************************************
function :	send command
parameter:
     Reg : Command register
******************************************************************************/
static void EPD_1IN54_A05_SendCommand(UBYTE Reg)
{
    DEV_Digital_Write(EPD_DC_PIN, 0);
    DEV_Digital_Write(EPD_CS_PIN, 0);
    DEV_SPI_WriteByte(Reg);
    DEV_Digital_Write(EPD_CS_PIN, 1);
}

/******************************************************************************
function :	send data
parameter:
    Data : Write data
******************************************************************************/
static void EPD_1IN54_A05_SendData(UBYTE Data)
{
    DEV_Digital_Write(EPD_DC_PIN, 1);
    DEV_Digital_Write(EPD_CS_PIN, 0);
    DEV_SPI_WriteByte(Data);
    DEV_Digital_Write(EPD_CS_PIN, 1);
}

static void EPD_1IN54_A05_SendData2(UBYTE *Data, UDOUBLE length)
{
    DEV_SPI_WriteByte(Data, length);
}

static void EPD_1IN54_A05_SendData2(UBYTE Data)
{
    // DEV_Digital_Write(EPD_DC_PIN, 1);
    // DEV_Digital_Write(EPD_CS_PIN, 0);
    DEV_SPI_WriteByte(Data);
    // DEV_Digital_Write(EPD_CS_PIN, 1);
}

/******************************************************************************
function :	Wait until the busy_pin goes LOW
parameter:
******************************************************************************/
void EPD_1IN54_A05_ReadBusy(void)
{
    Debug("e-Paper busy\r\n");
    while (DEV_Digital_Read(EPD_BUSY_PIN) == 1)
    { // LOW: idle, HIGH: busy
        DEV_Delay_ms(100);
    }
    Debug("e-Paper busy release\r\n");
}

/******************************************************************************
function :	Setting the display window
parameter:
******************************************************************************/
static void EPD_1IN54_A05_SetWindow(UWORD Xstart, UWORD Ystart, UWORD Xend, UWORD Yend)
{
    EPD_1IN54_A05_SendCommand(0x44); // SET_RAM_X_ADDRESS_START_END_POSITION
    EPD_1IN54_A05_SendData((Xstart >> 3) & 0xFF);
    EPD_1IN54_A05_SendData((Xend >> 3) & 0xFF);

    EPD_1IN54_A05_SendCommand(0x45); // SET_RAM_Y_ADDRESS_START_END_POSITION
    EPD_1IN54_A05_SendData(Ystart & 0xFF);
    EPD_1IN54_A05_SendData((Ystart >> 8) & 0xFF);
    EPD_1IN54_A05_SendData(Yend & 0xFF);
    EPD_1IN54_A05_SendData((Yend >> 8) & 0xFF);
}

/******************************************************************************
function :	Set Cursor
parameter:
******************************************************************************/
static void EPD_1IN54_A05_SetCursor(UWORD Xstart, UWORD Ystart)
{
    EPD_1IN54_A05_SendCommand(0x4E); // SET_RAM_X_ADDRESS_COUNTER
    EPD_1IN54_A05_SendData((Xstart >> 3) & 0xFF);

    EPD_1IN54_A05_SendCommand(0x4F); // SET_RAM_Y_ADDRESS_COUNTER
    EPD_1IN54_A05_SendData(Ystart & 0xFF);
    EPD_1IN54_A05_SendData((Ystart >> 8) & 0xFF);
}

/******************************************************************************
function :	Turn On Display
parameter:
******************************************************************************/
 void EPD_1IN54_A05_TurnOnDisplay(void)
{
    EPD_1IN54_A05_SendCommand(0x22); // DISPLAY_UPDATE_CONTROL_2
    EPD_1IN54_A05_SendData(0xC4);
    // EPD_1IN54_A05_SendData(0xC7);
    EPD_1IN54_A05_SendCommand(0x20); // MASTER_ACTIVATION
    EPD_1IN54_A05_SendCommand(0xFF); // TERMINATE_FRAME_READ_WRITE

    EPD_1IN54_A05_ReadBusy();
}

/******************************************************************************
function :	Initialize the e-Paper register
parameter:
******************************************************************************/
void EPD_1IN54_A05_Init(UBYTE Mode)
{
    EPD_1IN54_A05_Reset();

    EPD_1IN54_A05_SendCommand(0x01); // DRIVER_OUTPUT_CONTROL
    EPD_1IN54_A05_SendData((EPD_1IN54_A05_HEIGHT - 1) & 0xFF);
    EPD_1IN54_A05_SendData(((EPD_1IN54_A05_HEIGHT - 1) >> 8) & 0xFF);
    EPD_1IN54_A05_SendData(0x00); // GD = 0; SM = 0; TB = 0;

    // EPD_1IN54_A05_SendCommand(0x03); // VGH
    // EPD_1IN54_A05_SendData(0x00); //20V
    // EPD_1IN54_A05_SendCommand(0x04); //
    // EPD_1IN54_A05_SendData(0x0A);//17v
    // // EPD_1IN54_A05_SendData(0xCE);//8.8v
    // EPD_1IN54_A05_SendData(0x8E);//2.4v
    // EPD_1IN54_A05_SendData(0x3A);//-17v

    EPD_1IN54_A05_SendCommand(0x0C); // BOOSTER_SOFT_START_CONTROL
    // EPD_1IN54_A05_SendData(0xDE);
    // EPD_1IN54_A05_SendData(0xDD);
    // EPD_1IN54_A05_SendData(0x9D);
    EPD_1IN54_A05_SendData(0xD7);
    EPD_1IN54_A05_SendData(0xD6);
    EPD_1IN54_A05_SendData(0x9D);

    EPD_1IN54_A05_SendCommand(0x2C); // WRITE_VCOM_REGISTER
    EPD_1IN54_A05_SendData(0x77);    // VCOM 7C

    EPD_1IN54_A05_SendCommand(0x3A); // SET_DUMMY_LINE_PERIOD
    EPD_1IN54_A05_SendData(0x1A);    // 4 dummy lines per gate

    EPD_1IN54_A05_SendCommand(0x3B); // SET_GATE_TIME
    EPD_1IN54_A05_SendData(0x08);    // 2us per line


    EPD_1IN54_A05_SendCommand(0x11);
    EPD_1IN54_A05_SendData(0x03); // y+,x+

    // EPD_1IN54_A05_SendCommand(0x3C); //BorderWavefrom
    // EPD_1IN54_A05_SendData(0x33);   

    // set the look-up table register
    EPD_1IN54_A05_SendCommand(0x32);
    if (Mode == EPD_1IN54_A05_FULL)
    {
        for (UWORD i = 0; i < 30; i++)
        {
            EPD_1IN54_A05_SendData(EPD_1IN54_A05_lut_full_update[i]);
        }
    }
    else if (Mode == EPD_1IN54_A05_PART)
    {
        for (UWORD i = 0; i < 30; i++)
        {
            EPD_1IN54_A05_SendData(EPD_1IN54_A05_lut_partial_update[i]);
        }
    }
    else
    {
        Debug("error, the Mode is EPD_1IN54_A05_FULL or EPD_1IN54_A05_PART");
    }
}

/******************************************************************************
function :	Clear screen
parameter:
******************************************************************************/
void EPD_1IN54_A05_Clear(void)
{
    UWORD Width, Height;
    Width = (EPD_1IN54_A05_WIDTH % 8 == 0) ? (EPD_1IN54_A05_WIDTH / 8) : (EPD_1IN54_A05_WIDTH / 8 + 1);
    Height = EPD_1IN54_A05_HEIGHT;
    EPD_1IN54_A05_SetWindow(0, 0, EPD_1IN54_A05_WIDTH, EPD_1IN54_A05_HEIGHT);
    for (UWORD j = 0; j < Height; j++)
    {
        EPD_1IN54_A05_SetCursor(0, j);
        EPD_1IN54_A05_SendCommand(0x24);
        for (UWORD i = 0; i < Width; i++)
        {
            EPD_1IN54_A05_SendData(0XFF);
        }
    }
    EPD_1IN54_A05_TurnOnDisplay();
}

/******************************************************************************
function :	Sends the image buffer in RAM to e-Paper and displays
parameter:
******************************************************************************/
void EPD_1IN54_A05_Display(UBYTE *Image)
{
    UWORD Width, Height;
    Width = (EPD_1IN54_A05_WIDTH % 8 == 0) ? (EPD_1IN54_A05_WIDTH / 8) : (EPD_1IN54_A05_WIDTH / 8 + 1);
    Height = EPD_1IN54_A05_HEIGHT;

    UDOUBLE Addr = 0;
    // UDOUBLE Offset = ImageName;
    EPD_1IN54_A05_SetWindow(0, 0, EPD_1IN54_A05_WIDTH, EPD_1IN54_A05_HEIGHT);
    for (UWORD j = 0; j < Height; j++)
    {
        EPD_1IN54_A05_SetCursor(0, j);
        EPD_1IN54_A05_SendCommand(0x24);
        for (UWORD i = 0; i < Width; i++)
        {
            // Addr = i + j * Width;
            Addr = i + (j)*Width;
            EPD_1IN54_A05_SendData(Image[Addr]);
        }

        // EPD_1IN54_A05_SendCommand(0x26);//test
        // for (UWORD i = 0; i < Width; i++)
        // {
        //     EPD_1IN54_A05_SendData(0xFF);
        // }
    }
    // delay(100);
    EPD_1IN54_A05_TurnOnDisplay();
}

/******************************************************************************
function :	Enter sleep mode
parameter:
******************************************************************************/
void EPD_1IN54_A05_Sleep(void)
{
    EPD_1IN54_A05_SendCommand(0x10);
    EPD_1IN54_A05_SendData(0x01);
}

/******************************************************************************
function :	Sends the image buffer in RAM to e-Paper and displays
parameter:  color:black,0;red,1;*data;index;length
******************************************************************************/
void EPD_1IN54_A05_Data_Write(UBYTE color, UBYTE *Image_Data, UDOUBLE index, UDOUBLE length)
{
    if (index == 0)
    // if (1)
    {
        if (color == 0)
        {
            EPD_1IN54_A05_SetWindow(0, 0, EPD_1IN54_A05_WIDTH, EPD_1IN54_A05_HEIGHT);
            EPD_1IN54_A05_SetCursor(0, 0);
            EPD_1IN54_A05_SendCommand(0x24);
        }
        // else if (color == 1)
        // {
        //     EPD_1IN54_A05_SendCommand(0x26);
        // }
        DEV_Digital_Write(EPD_DC_PIN, 1);
    }

    UDOUBLE Width, Height;
    Width = (EPD_1IN54_A05_WIDTH % 8 == 0) ? (EPD_1IN54_A05_WIDTH / 8) : (EPD_1IN54_A05_WIDTH / 8 + 1);
    Height = EPD_1IN54_A05_HEIGHT;

    // if(length==(Width*Height))
    // {
    //     for (UDOUBLE j = 0; j < Height; j++)
    //     {
    //         for (UDOUBLE i = 0; i < Width; i++)
    //         {
    //             EPD_2IN9B_Z94_Grey_SendData(Image_Data[i + (Height-j) * Width]);
    //         }
    //     }

    // }
    // else{
#ifdef EPD_SPI_ENABLE
    EPD_1IN54_A05_SendData2(Image_Data, length);
#else
    for (UDOUBLE i = 0; i < length; i++)
    {
        if (color == 0)
        {
            // DEV_Digital_Write(EPD_DC_PIN, 1);
            // DEV_Digital_Write(EPD_CS_PIN, 0);
            EPD_1IN54_A05_SendData2(Image_Data[i]);
            // DEV_Digital_Write(EPD_CS_PIN, 1);
        }
        else if (color == 1)
        {
            // DEV_Digital_Write(EPD_DC_PIN, 1);
            // DEV_Digital_Write(EPD_CS_PIN, 0);
            EPD_1IN54_A05_SendData2(Image_Data[i]);
            // DEV_Digital_Write(EPD_CS_PIN, 1);
        }
    }
#endif
    // }
}


